#include "../mqserver/mq_exchange.hpp"
#include <gtest/gtest.h>

mymq::ExchangeManager::ptr emp;

class ExchangeTest : public testing::Environment
{
    public:
        virtual void SetUp() override
        {
            emp = std::make_shared<mymq::ExchangeManager>("./data/meta.db");
        }

        virtual void TearDown() override
        {
            //emp->clear();
        }
};

// TEST(exchange_test,insert_test)
// {
//     std::unordered_map<std::string,std::string> map = {{"k1","v1"},{"k2","v2"}};
//     emp->declareExchange("exchange1",mymq::ExchangeType::DIRECT,true,false,map);
//     emp->declareExchange("exchange2",mymq::ExchangeType::DIRECT,true,false,map);
//     emp->declareExchange("exchange3",mymq::ExchangeType::DIRECT,true,false,map);
//     emp->declareExchange("exchange4",mymq::ExchangeType::DIRECT,true,false,map);
//     ASSERT_EQ(emp->size(),4);
// }

TEST(exchange_test,select_test)
{
    ASSERT_EQ(emp->exists("exchange1"), true);
    ASSERT_EQ(emp->exists("exchange2"), false);
    ASSERT_EQ(emp->exists("exchange3"), true);
    ASSERT_EQ(emp->exists("exchange4"), true);
    mymq::Exchange::ptr exp = emp->selectExchange("exchange3");
    ASSERT_NE(exp.get(),nullptr);
    ASSERT_EQ(exp->name,"exchange3");
    ASSERT_EQ(exp->type,mymq::ExchangeType::DIRECT);
    ASSERT_EQ(exp->durable,true);
    ASSERT_EQ(exp->auto_delete,false);
    std::unordered_map<std::string,std::string> expected_args = {{"k1","v1"},{"k2","v2"}};
    ASSERT_EQ(exp->args, expected_args);
}

TEST(exchange_test,remove_test)
{
    emp->deleteExchange("exchange3");
    mymq::Exchange::ptr exp = emp->selectExchange("exchange3");
    ASSERT_EQ(exp.get(), nullptr);
    ASSERT_EQ(emp->exists("exchange2"), false);
}

int main(int argc,char* argv[])
{
    testing::InitGoogleTest(&argc,argv);
    testing::AddGlobalTestEnvironment(new ExchangeTest);
    RUN_ALL_TESTS();
    return 0;
}